package leetcode.editor.cn.dsa19_trie;
//设计一个使用单词列表进行初始化的数据结构，单词列表中的单词 互不相同 。 如果给出一个单词，请判定能否只将这个单词中一个字母换成另一个字母，使得所形成的新单
//词存在于你构建的字典中。 
//
// 实现 MagicDictionary 类：
// MagicDictionary() 初始化对象
// void buildDict(String[] dictionary) 使用字符串数组 dictionary 设定该数据结构，dictionary 中的字
//符串互不相同
// bool search(String searchWord) 给定一个字符串 searchWord ，判定能否只将字符串中 一个 字母换成另一个字母，使得
//所形成的新字符串能够与字典中的任一字符串匹配。如果可以，返回 true ；否则，返回 false 。
//
// 示例：
//输入
//["MagicDictionary", "buildDict", "search", "search", "search", "search"]
//[[], [["hello", "leetcode"]], ["hello"], ["hhllo"], ["hell"], ["leetcoded"]]
//输出
//[null, null, false, true, false, false]
//
//解释
//MagicDictionary magicDictionary = new MagicDictionary();
//magicDictionary.buildDict(["hello", "leetcode"]);
//magicDictionary.search("hello"); // 返回 False
//magicDictionary.search("hhllo"); // 将第二个 'h' 替换为 'e' 可以匹配 "hello" ，所以返回 True
//magicDictionary.search("hell"); // 返回 False
//magicDictionary.search("leetcoded"); // 返回 False
//
// 提示：
// 1 <= dictionary.length <= 100
// 1 <= dictionary[i].length <= 100
// dictionary[i] 仅由小写英文字母组成
// dictionary 中的所有字符串 互不相同
// 1 <= searchWord.length <= 100
// searchWord 仅由小写英文字母组成
// buildDict 仅在 search 之前调用一次
// 最多调用 100 次 search
//
// Related Topics 字典树 哈希表
// 👍 72 👎 0

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

public class ImplementMagicDictionary676_1 {
    //leetcode submit region begin(Prohibit modification and deletion)
    class MagicDictionary {
        Map<Integer, ArrayList<String>> wordMap;
        /**
         * Initialize your data structure here.
         */
        public MagicDictionary() {
            wordMap = new HashMap<>();
        }

        public void buildDict(String[] dictionary) {
            for (String word : dictionary) {
                // java8新特性:判断map中是否包含key,如果不包含，则将mapFunction(key)的值加⼊map中
                wordMap.computeIfAbsent(word.length(), x -> new ArrayList<>()).add(word);
            }
        }

        public boolean search(String searchWord) {
            // 先判断词的长度
            if (!wordMap.containsKey(searchWord.length())) return false;
            for (String possibleWord : wordMap.get(searchWord.length())) {
                // 查看是否存在长度差为1的字符
                if(countWordDiff(searchWord,possibleWord)) return true;
            }
            return false;
        }
        private boolean countWordDiff(String word1, String word2){
            int diff = 0;
            for (int i = 0; i < word1.length(); i++) {
                if (word1.charAt(i) != word2.charAt(i)) {
                    if (++diff > 1) return false;
                }
            }
            return diff == 1;
        }
    }
    /**
     * Your MagicDictionary object will be instantiated and called as such:
     * MagicDictionary obj = new MagicDictionary();
     * obj.buildDict(dictionary);
     * boolean param_2 = obj.search(searchWord);
     */
    //leetcode submit region end(Prohibit modification and deletion)
}